import { login, logout, getInfo } from "@/api/login";
import {
  getToken,
  setToken,
  removeToken,
  getSiteId,
  setSiteId,
  removeSiteId,
  setUserId,
  getUserId,
  removeUserId,
} from "@/utils/auth";
import defAva from "@/assets/images/Princess_ruoyi.png";
import { loginV2 } from "../../api/login";

const useUserStore = defineStore("user", {
  state: () => ({
    token: getToken(),
    SiteId: getSiteId(),
    UserId: getUserId(),
    id: "",
    name: "",
    avatar: "",
    roles: [],
    permissions: [],
  }),
  actions: {
    // 登录
    login(userInfo) {
      const username = userInfo.username.trim();
      const password = userInfo.password;
      const publicKey = userInfo.publicKey;
      return new Promise((resolve, reject) => {
        if (publicKey) {
          login(username, password, publicKey)
            .then((res) => {
              setToken(res.AccessToken);
              setSiteId(res.SiteId);
              setUserId(res.UserId);
              this.token = res.AccessToken;
              this.SiteId = res.SiteId;
              this.UserId = res.UserId;
              resolve(res);
            })
            .catch((error) => {
              reject(error);
            });
        } else {
          loginV2(username, password)
            .then((res) => {
              setToken(res.AccessToken);
              setSiteId(res.SiteId);
              setUserId(res.UserId);
              this.token = res.AccessToken;
              this.SiteId = res.SiteId;
              this.UserId = res.UserId;
              resolve(res);
            })
            .catch((error) => {
              reject(error);
            });
        }
      });
    },
    // 获取用户信息
    getInfo() {
      return new Promise((resolve, reject) => {
        getInfo()
          .then((res) => {
            const user = res.user;
            const avatar =
              user.avatar == "" || user.avatar == null
                ? defAva
                : import.meta.env.VITE_APP_BASE_API + user.avatar;

            if (res.roles && res.roles.length > 0) {
              // 验证返回的roles是否是一个非空数组
              this.roles = res.roles;
              this.permissions = res.permissions;
            } else {
              this.roles = ["ROLE_DEFAULT"];
            }
            this.id = user.userId;
            this.name = user.userName;
            this.avatar = avatar;
            resolve(res);
          })
          .catch((error) => {
            reject(error);
          });
      });
    },
    // 退出系统
    logOut() {
      return new Promise((resolve, reject) => {
        this.token = "";
        this.UserId = "";
        this.SiteId = "";
        this.roles = [];
        this.permissions = [];
        removeToken();
        removeUserId();
        removeSiteId();
        resolve();
      });
    },
  },
});

export default useUserStore;
